#encoding=utf-8
import random
from PIL import Image,ImageDraw,ImageFont,ImageFilter
from io import BytesIO

#生成几位数的验证码
number = 4
#生成验证码图片的高度和宽度
size = (129,53)
#背景颜色，默认为白色
bgcolor = (255,255,255)
#字体颜色，默认为蓝色
fontcolor = (0,0,0)
#干扰线颜色。默认为红色
linecolor = (0,0,0)
#是否要加入干扰线
draw_line = True
#加入干扰线条数的上下限
line_number = (1,5)

#用来随机生成一个字符串
def gene_text():

    source = ['1','2','3','4','5','6','7','8','9','A', 'B', 'C', 'D', 'E', 'F', 'G', 'H','I','J', 'K','L', 'M', 'N','P','Q','R','S', 'T', 'U', 'V', 'W', 'Z','X', 'Y']
    # 使用random.sample抽取4个样
    return ''.join(random.sample(source,number))#number是生成验证码的位数
#用来绘制干扰线
def gene_line(draw,width,height):

    begin = (0, random.randint(0, height)) # 起点
    end = (74, random.randint(0, height)) # 终点
    # 按照随机出来的起点和终点绘制一条线
    draw.line([begin, end], fill = linecolor,width=3)

#生成验证码
def gene_code():
    width,height = size #宽和高
    image = Image.new('RGBA',(width,height),bgcolor) #创建图片,设置他的颜色格式为RGB,尺寸大小,背景颜色
    import  os
    path = os.path.join(os.getcwd(),'Arial.ttf') # 指定字体文件

    font = ImageFont.truetype(path,36) #验证码的字体加载到图片文字上
    draw = ImageDraw.Draw(image)  #创建画笔
    text = gene_text() #生成字符串
    font_width, font_height = font.getsize(text)
    draw.text(((width - font_width) / number, (height - font_height) / number),text,\
            font= font,fill=fontcolor) #使用画笔,往图片上填充字符串
    if draw_line:
        gene_line(draw,width,height)
    image = image.transform((width+30,height+10), Image.AFFINE, (1,-0.3,0,-0.1,1,0),Image.BILINEAR)  #创建扭曲
    image = image.filter(ImageFilter.EDGE_ENHANCE_MORE) #滤镜，边界加强
    # image = image.filter(ImageFilter.EDGE_ENHANCE_MORE)  # 滤镜，边界加强
    bytes = BytesIO() # 把图片放到缓存中
    image.save(bytes,format='png')  # 保存验证码图片
    return bytes.getvalue(),text # 返回图片数据和文字,
